/*
 * Copyright (c) 2012 Robert Burrell Donkin robertburrelldonkin.name
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy of this software 
 * and associated documentation files (the "Software"), to deal in the Software without restriction, 
 * including without limitation the rights to use, copy, modify, merge, publish, distribute, 
 * sublicense, and/or sell copies of the Software, and to permit persons to whom the Software 
 * is furnished to do so, subject to the following conditions:
 * 
 * The above copyright notice and this permission notice shall be included in all copies or 
 * substantial portions of the Software.
 * 
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING 
 * BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND 
 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, 
 * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 * 
 */
package walrus.zip;

import java.io.File;
import java.io.IOException;
import java.util.Arrays;
import java.util.Random;

import walrus.base.ArticleBuffer;
import walrus.base.ArticleProcessor;

public class ArchiveArticleProcessor extends ArticleProcessor {
	
	private static final int DEFAULT_NUMBER_OF_ARCHIVES = 128;
	private final Random random;
	private final int numberOfArchives;
	private final ArchiveSaver[] archives;
	
	public ArchiveArticleProcessor(final File directory, final int numberOfArchives) {
		directory.mkdirs();
		archives = new ArchiveSaver[numberOfArchives];
		for (int i=0;i<numberOfArchives;i++) {
			archives[i] = new ArchiveSaver(new File(directory, "article-archive-" + i + ".zip"));
		}
		this.numberOfArchives = numberOfArchives;
		random = new Random();
	}
	
	public ArchiveArticleProcessor(String directoryName) {
		this(new File(directoryName), DEFAULT_NUMBER_OF_ARCHIVES);
	}

	@Override
	public ArticleBuffer process(ArticleBuffer buffer) throws Exception {
		final ArchiveSaver archive = archives[random.nextInt(numberOfArchives)];
		archive.save(buffer);
		return buffer;
	}
	
	@Override
	public void close() {
		for (final ArchiveSaver archive: archives) {
			try {
				archive.close();
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
	}

	@Override
	public String toString() {
		return "ArchiveArticleProcessor [archives=" + Arrays.toString(archives)
				+ ", numberOfArchives=" + numberOfArchives + ", random="
				+ random + "]";
	}
}
